DEVICE_ID=$1
RUN_MODE=$2
PRED_LEAD_TIME=${3:-1}
LR=${4:-0.001}
ROOT_DIR=$(pwd)
# export MS_ASCEND_CHECK_OVERFLOW_MODE=INFNAN_MODE
# export MINDSPORE_DUMP_CONFIG=/root/liulei/gc_ice_pred_0127/dump.json
export GLOG_v=3
LATENT_DIMES=512
PROCESSING_STEP=16
CURRENT_TIME=$(date -d @$(date +%s) "+%Y_%m_%d_%H_%M_%S")
GRAPH_PATH=${CURRENT_TIME}_${LATENT_DIMES}_${PROCESSING_STEP}
echo ${CURRENT_TIME}
LOG_FILE='all_summary_0321_pred_3_'${PRED_LEAD_TIME}
mkdir -p ${ROOT_DIR}/${LOG_FILE}/${GRAPH_PATH}
nohup python -u ./main.py \
  --device_id ${DEVICE_ID} \
  --run_mode ${RUN_MODE}\
  --mode GRAPH\
  --pred_lead_time ${PRED_LEAD_TIME}\
  --initial_lr ${LR}\
  --output_dir ${ROOT_DIR}/${LOG_FILE}/${GRAPH_PATH}\
  > ${ROOT_DIR}/${LOG_FILE}/${GRAPH_PATH}/train_${GRAPH_PATH}.log 2>&1 &


#   --save_graphs 0\